Update app.py

app.py

@@ -137,25 +137,16 @@ else:
             status_text.markdown("**📝 Generating report with Gemma...**")
             
             # Build the user content with query, sentiment results, and original review data.
-            user_content_text = (
-                "Generate a summary report based on this query and data:\n"
-                f"Query: {query_input}\n"
-                f"Sentiment Analysis Results: {scored_docs}\n"
-                f"Original Reviews: {candidate_docs}"
-            )
-            
-            messages = [
-                [
-                    #{
-                        #"role": "system",
-                        #"content": [{"type": "text", "text": "You are a helpful assistant."}]
-                    #},
-                    {
-                        "role": "user",
-                        "content": [{"type": "text", "text": user_content_text}]
-                    },
-                ],
-            ]
+            # Format the prompt as chat for Gemma
+                messages = [
+                    {"role": "user", "content": f"""
+            Generate a concise 300-word report based on the following analysis without repeating what's in the analysis.
+            Query:
+            "{query_input}"
+            Candidate Reviews with their scores:
+            {scored_docs}
+            """}
+                ]
             
             output = gemma_pipe(messages, max_new_tokens=50)
             progress_bar.progress(100)