refactor: improve response

app.py

@@ -167,24 +167,21 @@ def generate_answer(question, context, max_length=512):
     if model is None or tokenizer is None:
         return "Error: Could not load the model. Please try again later."
     
-    if not context or len(context.strip()) == 0:
-        return "No relevant autism research papers found to answer your question. Please try rephrasing your question."
-    
     # Format the context as a structured query
-    prompt = f"""Based on the following autism research papers, provide a detailed answer:
-
-Research Context:
-{context}
+    prompt = f"""You are an expert in autism research. Provide a comprehensive answer about autism, incorporating both general knowledge and specific research findings when available.
 
 Question: {question}
 
-Instructions: Provide a comprehensive answer that covers:
-1. Main findings about autism from the research
-2. Research methods used in autism studies
-3. Clinical implications for autism treatment
-4. Limitations of the autism studies
+Recent Research Context:
+{context}
+
+Instructions: Provide a detailed response that:
+1. Starts with a general overview of the topic as it relates to autism
+2. Incorporates specific findings from the provided research papers when relevant
+3. Discusses practical implications for individuals with autism and their families
+4. Mentions any limitations in current understanding
 
-If the research papers don't directly address autism and the question, explain what information is missing."""
+If the research papers don't directly address the question, focus on providing general, well-established information about autism while noting what specific research would be helpful."""
     
     try:
         # Generate response
@@ -194,7 +191,7 @@ If the research papers don't directly address autism and the question, explain w
             outputs = model.generate(
                 **inputs,
                 max_length=max_length,
-                min_length=100,
+                min_length=150,  # Increased minimum length for more comprehensive answers
                 num_beams=4,
                 length_penalty=1.5,
                 temperature=0.7,
@@ -204,15 +201,29 @@ If the research papers don't directly address autism and the question, explain w
         
         response = tokenizer.decode(outputs[0], skip_special_tokens=True)
         
-        # If response is too short or empty, provide a fallback message
-        if len(response.strip()) < 50:
-            return """I apologize, but I couldn't find specific information about autism in the research papers provided. 
-            This might be because:
-            1. The research papers don't directly address autism
-            2. The context needs more specific information about autism
-            3. The question might need to be more focused on autism-specific aspects
-            
-            Please try asking a more specific question about autism."""
+        # If response is too short or empty, provide a general overview
+        if len(response.strip()) < 100:
+            return f"""Here's what we know about autism in relation to your question about {question}:
+
+1. General Understanding:
+- Autism Spectrum Disorder (ASD) is a complex developmental condition
+- It affects how a person communicates, learns, and interacts with others
+- Each person with autism has unique strengths and challenges
+
+2. Key Aspects:
+- Communication and social interaction
+- Repetitive behaviors and specific interests
+- Sensory sensitivities
+- Early intervention is important
+
+3. Current Research:
+While the provided research papers don't directly address your specific question, researchers are actively studying various aspects of autism to better understand its causes, characteristics, and effective interventions.
+
+For more specific information, try asking about:
+- Specific symptoms or characteristics
+- Diagnostic processes
+- Treatment approaches
+- Current research in specific areas"""
         
         # Format the response for better readability
         formatted_response = response.replace(". ", ".\n").replace("• ", "\n• ")
@@ -231,7 +242,7 @@ This app searches through scientific papers to answer your questions about autis
 For best results, be specific in your questions.
 """)
 
-query = st.text_input("Please ask me anything about autism >")
+query = st.text_input("Please ask me anything about autism ✨")
 
 if query:
     with st.status("Searching for answers...") as status: