Update app.py

app.py

@@ -11,6 +11,7 @@ client=OpenAI(
 )
 
 model_name = 'gpt-3.5-turbo'
+rater_model = 'gpt-4o-mini'
 embedding_model = SentenceTransformerEmbeddings(model_name="thenlper/gte-large")
 persisted_vectordb_location = './vector_db/'
 collection_name = 'companies-10K-2023'
@@ -21,8 +22,38 @@ vectorstored_persisted = Chroma(
     persist_directory=persisted_vectordb_location
 )
 
-stored_documents = vectorstored_persisted.get(include=["metadatas"])
 
+def analyze_question(question)
+    status_button.update(value="Analyzing")
+    with open('./templates/question_analysis.txt', 'r') as file:
+        question_analysis = file.read()
+
+    with open('./templates/question_analysis_template.txt', 'r') as file:
+        question_analysis_template = file.read()
+
+    q_analysis = [
+        {"role": "system", "content": question_analysis},
+        {"role": "user", "content": question_template.format(
+            question=question,        
+            )
+        }
+    ]
+
+    try:
+        response = client.chat.completions.create(
+            model=model_name,  
+            messages=q_analysis,      
+            max_tokens=2000,
+            temperature=0.0
+        )
+
+        question_analysis_output.update(value=response.choices[0].message.content)
+        
+    except openai.OpenAIError as e:
+        status_button.update(value=f"An error occurred: {str(e)}")
+
+
+stored_documents = vectorstored_persisted.get(include=["metadatas"])
 source = set()
 document_names = set()
 
@@ -64,8 +95,7 @@ with gr.Blocks() as demo:
         
         with gr.Column(scale=1):
             status_button = gr.Button(value="Ready", interactive=False, elem_id="status_button")
-            
-            # Read-only textbox for Question Analysis
+                        
             question_analysis_output = gr.Textbox(
                 label="Question Analysis", 
                 placeholder="The analysis will be shown here...",
@@ -80,10 +110,9 @@ with gr.Blocks() as demo:
             )
     
     # Button click action to call the analyze_and_answer function
-    #analyze_button.click(
-    #    analyze_and_answer, 
-    #    inputs=[document_dropdown, temperature_slider, question_input], 
-    #    outputs=[question_analysis_output, final_answer_output]
-    #)
+    analyze_button.click(
+        analyze_question, 
+        inputs=question_input,
+    )
 
 demo.launch()