Update app.py

app.py

@@ -107,37 +107,32 @@ else:
                 html(timer(), height=50)
             status_text = st.empty()
             progress_bar = st.progress(0)
-            #try:
-                # Stage 1: Score candidate documents using the provided query.
-                status_text.markdown("**🔍 Scoring candidate documents...**")
-                progress_bar.progress(0)
-                
-                scored_docs = []
-                for doc in candidate_docs:
-                    combined_text = f"Query: {query_input} Document: {doc}"
-                    result = score_pipe(combined_text)[0]
-                    scored_docs.append((doc, result["score"]))
-                
-                progress_bar.progress(50)
-                
-                # Stage 2: Generate Report using Gemma, including query and scored results.
-                status_text.markdown("**📝 Generating report with Gemma...**")
-                prompt = f"""
-    Generate a detailed report based on the following analysis.
-    Query:
-    "{query_input}"
-    Candidate Reviews with their scores:
-    {scored_docs}
-    Please provide a concise summary report explaining the insights derived from these scores.
-    """
-                report = gemma_pipe(prompt, max_length=200)
-                progress_bar.progress(100)
-                status_text.success("**✅ Generation complete!**")
-                html("<script>localStorage.setItem('freezeTimer', 'true');</script>", height=0)
-                st.session_state.timer_frozen = True
-                st.write("**Scored Candidate Reviews:**", scored_docs)
-                st.write("**Generated Report:**", report[0]['generated_text'])
-            # except Exception as e:
-                #html("<script>document.getElementById('timer').remove();</script>")
-                s#tatus_text.error(f"**❌ Error:** {str(e)}")
-                #progress_bar.empty()
+            
+            # Stage 1: Score candidate documents (all reviews) without including the query.
+            status_text.markdown("**🔍 Scoring candidate documents...**")
+            progress_bar.progress(0)
+            
+            # Assuming score_pipe can take a list of texts directly:
+            scored_results = score_pipe(candidate_docs)
+            # Pair each review with its score assuming the output order matches the input order.
+            scored_docs = list(zip(candidate_docs, [result["score"] for result in scored_results]))
+            
+            progress_bar.progress(50)
+            
+            # Stage 2: Generate Report using Gemma, include the query and scored results.
+            status_text.markdown("**📝 Generating report with Gemma...**")
+            prompt = f"""
+Generate a detailed report based on the following analysis.
+Query:
+"{query_input}"
+Candidate Reviews with their scores:
+{scored_docs}
+Please provide a concise summary report explaining the insights derived from these scores.
+"""
+            report = gemma_pipe(prompt, max_length=200)
+            progress_bar.progress(100)
+            status_text.success("**✅ Generation complete!**")
+            html("<script>localStorage.setItem('freezeTimer', 'true');</script>", height=0)
+            st.session_state.timer_frozen = True
+            st.write("**Scored Candidate Reviews:**", scored_docs)
+            st.write("**Generated Report:**", report[0]['generated_text'])