Update app.py

app.py

@@ -178,6 +178,21 @@ def save_to_cosmos_db(container, query, response1, response2):
         unique_uuid = str(uuid.uuid4())
         return f"{timestamp}-{unique_uuid}"
 
+    def generate_short_filename(query):
+        # Create a hash of the query
+        hash_object = hashlib.md5(query.encode())
+        query_hash = hash_object.hexdigest()[:8]
+        
+        # Use the first few words of the query (up to 50 characters)
+        short_query = ' '.join(query.split()[:5])[:50]
+        
+        # Remove any non-alphanumeric characters and replace spaces with underscores
+        safe_query = ''.join(c if c.isalnum() else '_' for c in short_query)
+        
+        # Combine timestamp, hash, and safe query
+        timestamp = datetime.utcnow().strftime('%Y%m%d%H%M%S')
+        return f"{timestamp}_{query_hash}_{safe_query}.md"
+
     for attempt in range(max_retries):
         try:
             if container:
@@ -205,6 +220,13 @@ def save_to_cosmos_db(container, query, response1, response2):
                     query="SELECT * FROM c ORDER BY c._ts DESC",
                     enable_cross_partition_query=True
                 )
+
+                # Generate a short filename for local storage
+                filename = generate_short_filename(query)
+                with open(filename, 'w', encoding='utf-8') as file:
+                    file.write(f"Query: {query}\n\nResponse: {response1}")
+                st.success(f"File saved locally as: {filename}")
+
                 return
             else:
                 st.error("Cosmos DB container is not initialized.")
@@ -223,8 +245,6 @@ def save_to_cosmos_db(container, query, response1, response2):
 
     st.error("Failed to save record after maximum retries.")
 
-    
-
 # Add dropdowns for model and database choices
 def search_glossary(query):
     st.markdown(f"### 🔍 Search Glossary for: `{query}`")