Update app.py via AI Editor

app.py

@@ -12,15 +12,16 @@ from PyPDF2 import PdfReader
 import logging
 import threading
 import re
-import google.generativeai as genai
+import openai
 import sqlite3
 import time
 
 logging.basicConfig(level=logging.INFO, format='%(asctime)s %(levelname)s %(message)s')
 
-GEMINI_KEY = os.environ.get("GEMINI_KEY", "")
-genai.configure(api_key=GEMINI_KEY)
-GEMINI_MODEL = "gemini-2.5-pro-preview-03-25"
+OPENAI_KEY = os.environ.get("OPENAI_API_KEY", "")
+openai.api_key = OPENAI_KEY
+OPENAI_MODEL = "gpt-4-1106-preview"
+OPENAI_MAX_TOKENS = 32768
 
 DB_PATH = "maiko_dash_docs.sqlite"
 
@@ -645,23 +646,31 @@ def unified_handler(contents_lists, n_clicks_generate_list, n_clicks_download, s
         else:
             context += f"\n\n{inputs.get('shred','')}"
         full_prompt = f"{prompt}\n\n{context}"
-        logging.info(f"Prompt sent to Gemini (first 1000 chars): {full_prompt[:1000]}")
+        logging.info(f"Prompt sent to OpenAI (first 1000 chars): {full_prompt[:1000]}")
         result_holder = {}
 
-        def gemini_thread():
+        def openai_thread():
             try:
-                model = genai.GenerativeModel(GEMINI_MODEL)
-                response = model.generate_content(full_prompt, stream=False)
-                result_holder['result'] = response.text if hasattr(response, 'text') else str(response)
-                logging.info("Gemini document generated successfully.")
+                response = openai.chat.completions.create(
+                    model=OPENAI_MODEL,
+                    messages=[
+                        {"role": "system", "content": "You are a helpful assistant for proposal writing. Always output in markdown, and use markdown tables for spreadsheets if needed."},
+                        {"role": "user", "content": full_prompt}
+                    ],
+                    max_tokens=OPENAI_MAX_TOKENS,
+                    temperature=0.3,
+                )
+                result_text = response.choices[0].message.content if hasattr(response.choices[0].message, "content") else str(response)
+                result_holder['result'] = result_text
+                logging.info("OpenAI document generated successfully.")
             except Exception as e:
-                logging.error(f"Error with Gemini API: {str(e)}")
+                logging.error(f"Error with OpenAI API: {str(e)}")
                 result_holder['result'] = f"Error generating document: {str(e)}"
-        t = threading.Thread(target=gemini_thread)
+        t = threading.Thread(target=openai_thread)
         t.start()
-        t.join(timeout=60)
+        t.join(timeout=120)
         generated_doc = result_holder.get('result', 'Error: No document generated.')
-        logging.info(f"Gemini API returned: {generated_doc[:500]}")  # LOGGING LINE
+        logging.info(f"OpenAI API returned: {generated_doc[:500]}")  # LOGGING LINE
 
         set_generated_doc(selected_type, generated_doc)
         # Preview with fallback
@@ -746,5 +755,7 @@ def unified_handler(contents_lists, n_clicks_generate_list, n_clicks_download, s
 
     return stores + [file_list, uploaded_doc_name_shred, preview, dash.no_update, dash.no_update, loader_trigger]
 
-if __name__ == "__main__":
-    app.run_server(debug=True)
+if __name__ == '__main__':
+    print("Starting the Dash application...")
+    app.run(debug=True, host='0.0.0.0', port=7860)
+    print("Dash application has finished running.")