Update app.py (#2)

- Update app.py (01bb8350fb1c7f9e763e88f26e0b9d304ee87b96)


Co-authored-by: Aryan Wadhawan <aryanxxvii@users.noreply.huggingface.co>

app.py

@@ -96,8 +96,10 @@ def get_relevant_context(query):
         
         # Extract and combine the content from retrieved documents
         context = "\n".join([doc.page_content for doc in docs])
+        context = "Use the following medical context to inform your response: " + context
+        
+        return context if not context else "" 
         
-        return context
     except Exception as e:
         print(f"Error in similarity search: {e}")
         return "Could not retrieve relevant context."
@@ -105,7 +107,7 @@ def get_relevant_context(query):
 # Update system prompt to include retrieved context
 def get_enhanced_prompt(query, context):
     enhanced_prompt = f"""You have to act as a professional doctor, i know you are not but this is for learning purpose.
-    Use the following medical context to inform your response: {context}
+    {context}
     What's in this image?. Do you find anything wrong with it medically? 
     If you make a differential, suggest some remedies for them. Donot add any numbers or special characters in 
     your response. Your response should be in one long paragraph. Also always answer as if you are answering to a real person.