Update app.py

app.py

@@ -12,6 +12,7 @@ import json
 import uuid  # 🎲 For generating unique IDs
 from urllib.parse import quote  # 🔗 For encoding URLs
 from gradio_client import Client  # 🌐 For connecting to Gradio apps
+import openai  # 🤖 For OpenAI API interactions
 
 # 🎉 Welcome to our fun-filled Cosmos DB and GitHub Integration app!
 st.set_page_config(layout="wide")
@@ -25,6 +26,10 @@ Key = os.environ.get("Key")  # 🔑 Don't forget your key!
 # 🏠 Your local app URL (Change this to your app's URL)
 LOCAL_APP_URL = "http://localhost:8501"
 
+# 🤖 OpenAI configuration
+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):
     # 🚚 Let's download that GitHub repo!
@@ -182,6 +187,77 @@ def search_glossary(query):
     st.markdown(responseall)
     return responseall
 
+# 📝 Function to process text input
+def process_text(text_input):
+    if text_input:
+        if 'messages' not in st.session_state:
+            st.session_state.messages = []
+        
+        st.session_state.messages.append({"role": "user", "content": text_input})
+
+        with st.chat_message("user"):
+            st.markdown(text_input)
+            
+        with st.chat_message("assistant"):
+            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
+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)
+    safe_text = "_".join(safe_text.strip().split())
+    filename = f"{safe_text}.{file_type}"
+    return filename
+
+# 🏷️ Function to extract markdown title
+def extract_markdown_title(content):
+    # 🔍 Extract the first markdown heading as the title
+    lines = content.splitlines()
+    for line in lines:
+        if line.startswith('#'):
+            return line.lstrip('#').strip()
+    return None
+
+# 💾 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.
+    """
+    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)
+    if file_type == "md":
+        title_from_content = extract_markdown_title(content)
+        if title_from_content:
+            filename = generate_filename(title_from_content, file_type)
+
+    # Step 3: Save the file
+    with open(filename, "w", encoding="utf-8") as f:
+        if is_image:
+            f.write(content)
+        else:
+            f.write(prompt + "\n\n" + content)
+
+    return filename
+
 # 🎈 Let's modify the main app to be more fun!
 def main():
     st.title("🐙Git🌌Cosmos💫 - Azure Cosmos DB and Github Agent")
@@ -204,13 +280,16 @@ def main():
     if 'cloned_doc' not in st.session_state:
         st.session_state.cloned_doc = None
 
-    # 🔍 Check for query parameters
-    query_params = st.get_query_params()
-    if 'q' in query_params:
-        query = query_params['q'][0]  # Get the first value of 'q'
-        # 🕵️‍♂️ We have a query! Let's search the glossary!
-        search_glossary(query)
-        st.stop()  # Stop further execution
+    # ⚙️ q= Run ArXiv search from query parameters
+    try:
+        query_params = st.query_params
+        query = (query_params.get('q') or query_params.get('query') or [''])[0]
+        if query:
+            # 🕵️‍♂️ We have a query! Let's process it!
+            process_text(query)
+            st.stop()  # Stop further execution
+    except Exception as e:
+        st.markdown(' ')
 
     # 🔐 Automatic Login
     if Key:
@@ -300,10 +379,10 @@ def main():
 
                             # 🔗 Let's create a list of links for these values
                             search_urls = {
-                                "🚀🌌ArXiv": lambda k: f"/?q={quote(k)}",
-                                "🃏Analyst": lambda k: f"/?q={quote(k)}-{quote('PromptPrefix')}",
-                                "📚PyCoder": lambda k: f"/?q={quote(k)}-{quote('PromptPrefix2')}",
-                                "🔬JSCoder": lambda k: f"/?q={quote(k)}-{quote('PromptPrefix3')}",
+                                "🚀🌌ArXiv": lambda k: f"{LOCAL_APP_URL}/?q={quote(k)}",
+                                "🃏Analyst": lambda k: f"{LOCAL_APP_URL}/?q={quote(k)}-{quote('PromptPrefix')}",
+                                "📚PyCoder": lambda k: f"{LOCAL_APP_URL}/?q={quote(k)}-{quote('PromptPrefix2')}",
+                                "🔬JSCoder": lambda k: f"{LOCAL_APP_URL}/?q={quote(k)}-{quote('PromptPrefix3')}",
                                 "🏠": lambda k: f"{LOCAL_APP_URL}/?q={quote(k)}",
                                 "📖": lambda k: f"https://en.wikipedia.org/wiki/{quote(k)}",
                                 "🔍": lambda k: f"https://www.google.com/search?q={quote(k)}",
@@ -497,8 +576,8 @@ def main():
                         finally:
                             if os.path.exists(local_path):
                                 shutil.rmtree(local_path)
-                        #else:
-                            #st.error("Please ensure GitHub token is set in environment variables and source repository URL is provided. 🔑❓")
+                        else:
+                            st.error("Please ensure GitHub token is set in environment variables and source repository URL is provided. 🔑❓")
     
         except exceptions.CosmosHttpResponseError as e:
             st.error(f"Failed to connect to Cosmos DB. HTTP error: {str(e)} 🚨")