Update app.py

app.py

@@ -137,6 +137,56 @@ def process_upload(api_key, files, lang):
     else:
         return f"No valid documents were indexed. Errors: {'; '.join(error_messages)}", None
 
+# This is the missing query_app function that needs to be defined
+def query_app(query, model_name, use_similarity_check, openai_api_key):
+    global vector_index, query_log
+
+    if vector_index is None:
+        logging.error("No documents indexed yet. Please upload documents first.")
+        return "No documents indexed yet. Please upload documents first.", None
+
+    if not openai_api_key:
+        logging.error("No OpenAI API Key provided.")
+        return "Please provide a valid OpenAI API Key.", None
+
+    try:
+        llm = OpenAI(model=model_name, api_key=openai_api_key)
+    except Exception as e:
+        logging.error(f"Error initializing the OpenAI model: {e}")
+        return f"Error initializing the OpenAI model: {e}", None
+
+    response_synthesizer = get_response_synthesizer(llm=llm)
+    query_engine = vector_index.as_query_engine(llm=llm, response_synthesizer=response_synthesizer)
+
+    try:
+        response = query_engine.query(query)
+    except Exception as e:
+        logging.error(f"Error during query processing: {e}")
+        return f"Error during query processing: {e}", None
+
+    generated_response = response.response
+    query_log.append({
+        "query_id": str(len(query_log) + 1),
+        "query": query,
+        "gt_answer": "Placeholder ground truth answer",
+        "response": generated_response,
+        "retrieved_context": [{"text": doc.text} for doc in response.source_nodes]
+    })
+
+    metrics = {}
+
+    if use_similarity_check:
+        try:
+            logging.info("Similarity check is enabled. Calculating similarity.")
+            similarity = calculate_similarity(generated_response, "Placeholder ground truth answer")
+            metrics['similarity'] = similarity
+            logging.info(f"Similarity calculated: {similarity}")
+        except Exception as e:
+            logging.error(f"Error during similarity calculation: {e}")
+            metrics['error'] = f"Error during similarity calculation: {e}"
+
+    return generated_response, metrics if use_similarity_check else None
+
 def main():
     with gr.Blocks(title="Document Processing App") as demo:
         gr.Markdown("# 📄 Document Processing and Querying App")
@@ -167,14 +217,16 @@ def main():
                     value="gpt-4o",
                     label="Select Model"
                 )
+                similarity_checkbox = gr.Checkbox(label="Use Similarity Check", value=False)
                 query_button = gr.Button("Ask")
             with gr.Column():
                 answer_output = gr.Textbox(label="Answer", interactive=False)
+                metrics_output = gr.JSON(label="Metrics")
 
             query_button.click(
                 fn=query_app,
-                inputs=[query_input, model_dropdown, api_key_input],
-                outputs=[answer_output]
+                inputs=[query_input, model_dropdown, similarity_checkbox, api_key_input],
+                outputs=[answer_output, metrics_output]
             )
 
         gr.Markdown("""