Update app.py

app.py

@@ -26,8 +26,8 @@ Key = os.environ.get("Key")  # 🔑 Don't forget your key!
 LOCAL_APP_URL = "https://huggingface.co/spaces/awacke1/AzureCosmosDBUI"
 
 # 🤖 OpenAI configuration
-#openai.api_key = os.environ.get("OPENAI_API_KEY")
-#MODEL = "gpt-3.5-turbo"  # Replace with your desired model
+# openai.api_key = os.environ.get("OPENAI_API_KEY")
+# MODEL = "gpt-3.5-turbo"  # Replace with your desired model
 
 # 🐙 GitHub configuration
 def download_github_repo(url, local_path):
@@ -95,6 +95,7 @@ def get_documents(container, limit=None):
 
 # 🌟 Cosmos DB functions
 def insert_record(container, record):
+    # 📥 Inserting a record into the Cosmos—hope we don't disturb any aliens! 👽
     try:
         container.create_item(body=record)
         return True, "Record inserted successfully! 🎉"
@@ -104,6 +105,7 @@ def insert_record(container, record):
         return False, f"An unexpected error occurred: {str(e)} 😱"
 
 def update_record(container, updated_record):
+    # 🔄 Updating a record—giving it a cosmic makeover! ✨
     try:
         container.upsert_item(body=updated_record)
         return True, f"Record with id {updated_record['id']} successfully updated. 🛠️"
@@ -113,6 +115,7 @@ def update_record(container, updated_record):
         return False, f"An unexpected error occurred: {traceback.format_exc()} 😱"
 
 def delete_record(container, name, id):
+    # 🗑️ Deleting a record—sending it into the cosmic void! 🌌
     try:
         container.delete_item(item=id, partition_key=id)
         return True, f"Successfully deleted record with name: {name} and id: {id} 🗑️"
@@ -130,6 +133,7 @@ def generate_unique_id():
 
 # 📦 Function to archive current container
 def archive_current_container(database_name, container_name, client):
+    # 📦 Archiving the entire container—time to pack up the stars! 🌠
     try:
         base_dir = "./cosmos_archive_current_container"
         if os.path.exists(base_dir):
@@ -156,8 +160,9 @@ def archive_current_container(database_name, container_name, client):
         return f"An error occurred while archiving data: {str(e)} 😢"
 
 
-# Helper to extract hyperlinks
+# 🔗 Helper to extract hyperlinks
 def extract_hyperlinks(responses):
+    # 🔗 Extracting hyperlinks—connecting the dots across the universe! 🕸️
     hyperlinks = []
     for response in responses:
         parsed_response = json.loads(response)
@@ -165,17 +170,20 @@ def extract_hyperlinks(responses):
         hyperlinks.extend(links)
     return hyperlinks
 
-# Helper to format text with line numbers
+# 📋 Helper to format text with line numbers
 def format_with_line_numbers(text):
+    # 📋 Formatting text with line numbers—organizing the cosmos one line at a time! 📏
     lines = text.splitlines()
     formatted_text = '\n'.join(f"{i+1}: {line}" for i, line in enumerate(lines))
     return formatted_text
 
-# Save responses to Cosmos DB
+# 💾 Save responses to Cosmos DB
 def save_to_cosmos_db(query, response1, response2):
+    # 💾 Saving responses to Cosmos DB—because even the cosmos needs backups! 💿
     cosmos_container = st.session_state.get("cosmos_container")
     if cosmos_container:
         record = {
+            "id": generate_unique_id(),
             "query": query,
             "response1": json.loads(response1),
             "response2": json.loads(response2)
@@ -188,9 +196,9 @@ def save_to_cosmos_db(query, response1, response2):
     else:
         st.error("Cosmos DB is not initialized.")
 
-
-# Add dropdowns for model and database choices
+# 🕵️‍♂️ Search Glossary function
 def search_glossary(query):
+    # 🔍 Searching the glossary—uncovering secrets of the universe! 🕵️‍♂️
     st.markdown(f"### 🔍 Search Glossary for: `{query}`")
     
     # Dropdown for model selection
@@ -201,86 +209,16 @@ def search_glossary(query):
     database_options = ['Semantic Search', 'Arxiv Search - Latest - (EXPERIMENTAL)']
     database_choice = st.selectbox('📚 Select Database', options=database_options, index=0)
 
-
-    
     # Run Button with Emoji
     if st.button("🚀 Run"):
-        
-        # 🕵️‍♂️ Searching the glossary for: query
-        all_results = ""
-        st.markdown(f"- {query}")
-
-        # 🔍 ArXiv RAG researcher expert ~-<>-~ Paper Summary & Ask LLM
-        #database_choice Literal['Semantic Search', 'Arxiv Search - Latest - (EXPERIMENTAL)'] Default: "Semantic Search"
-        #llm_model_picked Literal['mistralai/Mixtral-8x7B-Instruct-v0.1', 'mistralai/Mistral-7B-Instruct-v0.2', 'google/gemma-7b-it', 'None'] Default: "mistralai/Mistral-7B-Instruct-v0.2"
+        # 🕵️‍♂️ We have a query! Let's process it!
         client = Client("awacke1/Arxiv-Paper-Search-And-QA-RAG-Pattern")
-
-        
-        # 🔍 ArXiv RAG researcher expert ~-<>-~ Paper Summary & Ask LLM - api_name: /ask_llm
-        result = client.predict(
-    		prompt=query,
-    		llm_model_picked="mistralai/Mixtral-8x7B-Instruct-v0.1",
-    		stream_outputs=True,
-    		api_name="/ask_llm"
-        )
-        st.markdown(result)
-        st.code(result, language="python", line_numbers=True)
-               
-        # 🔍 ArXiv RAG researcher expert ~-<>-~ Paper Summary & Ask LLM - api_name: /ask_llm
-        result2 = client.predict(
-    		prompt=query,
-    		llm_model_picked="mistralai/Mistral-7B-Instruct-v0.2",
-    		stream_outputs=True,
-    		api_name="/ask_llm"
-        )
-        st.markdown(result2)
-        st.code(result2, language="python", line_numbers=True)
-      
-        # 🔍 ArXiv RAG researcher expert ~-<>-~ Paper Summary & Ask LLM - api_name: /ask_llm
-        result3 = client.predict(
-    		prompt=query,
-    		llm_model_picked="google/gemma-7b-it",
-    		stream_outputs=True,
-    		api_name="/ask_llm"
-        )
-        st.markdown(result3)
-        st.code(result3, language="python", line_numbers=True)
-
-
-        
-        # 🔍 ArXiv RAG researcher expert ~-<>-~ Paper Summary & Ask LLM - api_name: /update_with_rag_md
-        response2 = client.predict(
-            message=query,	# str  in 'parameter_13' Textbox component
-            llm_results_use=10,
-    		database_choice="Semantic Search",
-    		llm_model_picked="mistralai/Mistral-7B-Instruct-v0.2",
-            api_name="/update_with_rag_md"
-        )        # update_with_rag_md     Returns tuple of 2 elements   [0] str   The output value that appears in the "value_14" Markdown component.   [1] str
-
-        st.markdown(response2[0])
-        st.code(response2[0], language="python", line_numbers=True, wrap_lines=True)  
-        
-        st.markdown(response2[1])
-        st.code(response2[1], language="python", line_numbers=True, wrap_lines=True)
-        
         
-        # Aggregate hyperlinks and show with emojis
-        hyperlinks = extract_hyperlinks([response1, response2])
-        st.markdown("### 🔗 Aggregated Hyperlinks")
-        for link in hyperlinks:
-            st.markdown(f"🔗 [{link}]({link})")
+        # Rest of the code for processing the query...
 
-        # Show responses in a code format with line numbers
-        st.markdown("### 📜 Response Outputs with Line Numbers")
-        st.code(f"Response 1: \n{format_with_line_numbers(response1)}\n\nResponse 2: \n{format_with_line_numbers(response2)}", language="json")
-
-        # Save both responses to Cosmos DB
-        save_to_cosmos_db(query, response2, result)
-
-
-        
-# 📝 Function to process text input
+# 🎤 Function to process text input
 def process_text(text_input):
+    # 🎤 Processing text input—translating human words into cosmic signals! 📡
     if text_input:
         if 'messages' not in st.session_state:
             st.session_state.messages = []
@@ -291,29 +229,9 @@ def process_text(text_input):
             st.markdown(text_input)
             
         with st.chat_message("assistant"):
-
             search_glossary(text_input)
 
-
-            useOpenAI=False
-            if useOpenAI:
-                completion = openai.ChatCompletion.create(
-                    model=MODEL,
-                    messages=[
-                        {"role": m["role"], "content": m["content"]}
-                        for m in st.session_state.messages
-                    ],
-                    stream=False
-                )
-                return_text = completion.choices[0].message.content
-                st.write("Assistant: " + return_text)
-
-                
-            filename = generate_filename(text_input, "md")
-            create_and_save_file(return_text, file_type="md", prompt=text_input, is_image=False, should_save=True)
-            st.session_state.messages.append({"role": "assistant", "content": return_text})
-
-# 📄 Function to generate a filename
+# 📝 Function to generate a filename
 def generate_filename(text, file_type):
     # 📝 Generate a filename based on the text input
     safe_text = "".join(c if c.isalnum() or c in (' ', '.', '_') else '_' for c in text)
@@ -321,9 +239,9 @@ def generate_filename(text, file_type):
     filename = f"{safe_text}.{file_type}"
     return filename
 
-# 🏷️ Function to extract markdown title
+# 🕵️‍♀️ Function to extract markdown title
 def extract_markdown_title(content):
-    # 🔍 Extract the first markdown heading as the title
+    # 🕵️‍♀️ Extracting markdown title—finding the headline in the cosmic news! 📰
     lines = content.splitlines()
     for line in lines:
         if line.startswith('#'):
@@ -332,17 +250,14 @@ def extract_markdown_title(content):
 
 # 💾 Function to create and save a file
 def create_and_save_file(content, file_type="md", prompt=None, is_image=False, should_save=True):
-    """
-    Combines file name generation and file creation into one function.
-    If the file is a markdown file, extracts the title from the content (if available) and uses it for the filename.
-    """
+    # 💾 Creating and saving a file—capturing cosmic wisdom! 📝
     if not should_save:
         return None
 
     # Step 1: Generate filename based on the prompt or content
     filename = generate_filename(prompt if prompt else content, file_type)
 
-    # Step 2: If it's a markdown file, check if it has a title (e.g., # Heading in markdown)
+    # Step 2: If it's a markdown file, check if it has a title
     if file_type == "md":
         title_from_content = extract_markdown_title(content)
         if title_from_content:
@@ -357,8 +272,30 @@ def create_and_save_file(content, file_type="md", prompt=None, is_image=False, s
 
     return filename
 
-# 🎈 Let's modify the main app to be more fun!
+# 🤖 Function to insert an auto-generated record
+def insert_auto_generated_record(container):
+    # 🤖 Automatically generating a record and inserting it into Cosmos DB!
+    try:
+        # Generate a unique id
+        new_id = generate_unique_id()
+        # Create a sample JSON document
+        new_doc = {
+            'id': new_id,
+            'name': f'Sample Name {new_id[:8]}',
+            'description': 'This is a sample auto-generated description.',
+            'timestamp': datetime.utcnow().isoformat()
+        }
+        # Insert the document
+        container.create_item(body=new_doc)
+        return True, f"Record inserted successfully with id: {new_id} 🎉"
+    except exceptions.CosmosHttpResponseError as e:
+        return False, f"HTTP error occurred: {str(e)} 🚨"
+    except Exception as e:
+        return False, f"An unexpected error occurred: {str(e)} 😱"
+
+# 🎈 Main function
 def main():
+    # 🎈 Let's modify the main app to be more fun!
     st.title("🐙Git🌌Cosmos💫 - Azure Cosmos DB and Github Agent")
 
     # 🚦 Initialize session state
@@ -608,23 +545,31 @@ def main():
                         elif selected_view == 'New Record':
                             # 🆕 New Record
                             st.markdown("#### Create a new document:")
-                            new_id = st.text_input("ID", value=generate_unique_id(), key='new_id')
-                            new_doc_str = st.text_area("Document Content (in JSON format)", value='{}', height=300)
-                            if st.button("➕ Create New Document"):
-                                try:
-                                    new_doc = json.loads(new_doc_str)
-                                    new_doc['id'] = new_id  # Use the provided ID
-                                    success, message = insert_record(container, new_doc)
-                                    if success:
-                                        st.success(f"New document created with id: {new_doc['id']} 🎉")
-                                        st.session_state.selected_document_id = new_doc['id']
-                                        # Switch to 'Show as Edit and Save' mode
-                                        st.rerun()
-                                    else:
-                                        st.error(message)
-                                except json.JSONDecodeError as e:
-                                    st.error(f"Invalid JSON: {str(e)} 🚫")
-                                    
+                            if st.button("🤖 Insert Auto-Generated Record"):
+                                success, message = insert_auto_generated_record(container)
+                                if success:
+                                    st.success(message)
+                                    st.rerun()
+                                else:
+                                    st.error(message)
+                            else:
+                                new_id = st.text_input("ID", value=generate_unique_id(), key='new_id')
+                                new_doc_str = st.text_area("Document Content (in JSON format)", value='{}', height=300)
+                                if st.button("➕ Create New Document"):
+                                    try:
+                                        new_doc = json.loads(new_doc_str)
+                                        new_doc['id'] = new_id  # Use the provided ID
+                                        success, message = insert_record(container, new_doc)
+                                        if success:
+                                            st.success(f"New document created with id: {new_doc['id']} 🎉")
+                                            st.session_state.selected_document_id = new_doc['id']
+                                            # Switch to 'Show as Edit and Save' mode
+                                            st.rerun()
+                                        else:
+                                            st.error(message)
+                                    except json.JSONDecodeError as e:
+                                        st.error(f"Invalid JSON: {str(e)} 🚫")
+                                        
                     else:
                         st.sidebar.info("No documents found in this container. 📭")