Merge branch 'main' of https://huggingface.co/spaces/TeamTonic/MultiMed

app.py

@@ -230,6 +230,32 @@ def convert_to_markdown(vectara_response_json):
         return "No data found in the response."
 # Main function to handle the Gradio interface logic
 
+def process_summary_with_openai(summary):
+    """
+    This function takes a summary text as input and processes it with OpenAI's GPT model.
+    """
+    try:
+        # Ensure that the OpenAI client is properly initialized
+        client = OpenAI(api_key=os.getenv('OPENAI_API_KEY'))
+        
+        # Create the prompt for OpenAI's completion
+        prompt = "You are clinical consultant discussion training cases with students at TonicUniversity. You will recieve a summary assessment. Assess and describe the proper options in minute detail. Propose a course of action based on your assessment. Exclude any other commentary:"
+        
+        # Call the OpenAI API with the prompt and the summary
+        completion = client.chat.completions.create(
+            model="gpt-4-1106-preview",  # Make sure to use the correct model name
+            messages=[
+                {"role": "system", "content": prompt},
+                {"role": "user", "content": summary}
+            ]
+        )
+        
+        # Extract the content from the completion
+        final_summary = completion.choices[0].message.content
+        return final_summary
+    except Exception as e:
+        return str(e)
+        
 
 def process_and_query(text=None):
     try:
@@ -245,12 +271,20 @@ def process_and_query(text=None):
         #     text = "the user asks the following to his health adviser " + text
             
 
-        # Now, use the text (either provided by the user or obtained from OpenAI) to query Vectara
+        
+        # Use the text to query Vectara
         vectara_response_json = query_vectara(text)
+        
+        # Convert the Vectara response to Markdown
         markdown_output = convert_to_markdown(vectara_response_json)
-        client = OpenAI()
-        prompt ="Answer in the same language, write it better,remove the sources, more understandable and make it 2 line answer:"
-        markdown_output_final = markdown_output
+        
+        # Process the summary with OpenAI
+        final_response = process_summary_with_openai(markdown_output)
+        
+        # Return the processed summary along with the full output
+        return f"**Summary**: {final_response}\n\n**Full output**:\n{markdown_output}"
+    except Exception as e:
+        return str(e)
 
         completion = client.chat.completions.create(
           model="gpt-3.5-turbo",