Update app.py

app.py

@@ -148,7 +148,9 @@ def create_bibtex_entry(data):
     bibtex = bibtex.rstrip(',\n') + "\n}"
     return bibtex
 
-def save_bibtex(bibtex_content):
+def save_bibtex(combined_content):
+    # Extract BibTeX content (remove the HTML paragraph)
+    bibtex_content = re.sub(r'<p>.*?</p>\s*', '', combined_content, flags=re.DOTALL)
     with tempfile.NamedTemporaryFile(mode='w', delete=False, suffix='.bib') as temp_file:
         temp_file.write(bibtex_content)
     return temp_file.name
@@ -209,7 +211,7 @@ class CombinedProcessor:
 
         list_style = pd.concat(list_style)
         list_style = list_style.groupby('label')['score'].mean().sort_values(ascending=False).reset_index()
-        top_style = result.iloc[0]['label']
+        top_style = list_style.iloc[0]['label']
         print(top_style)
         
         #Disambiguation to avoid duplicate ids.
@@ -217,7 +219,15 @@ class CombinedProcessor:
         
         # Join BibTeX entries with HTML formatting
         formatted_entries = [html.escape(entry) for entry in bibtex_entries]
-        return "\n\n".join(formatted_entries)
+
+        bibtex_content = "\n\n".join(formatted_entries)
+
+        # Create a paragraph with the top style information
+        style_info = f"<p>Top bibliography style: {top_style} (Mean score: {top_style_score:.6f})</p>"
+
+        # Combine the style information and BibTeX content
+        result = f"{style_info}\n\n{bibtex_content}"
+        return result
 
 # Create the processor instance
 processor = CombinedProcessor()
@@ -227,7 +237,7 @@ with gr.Blocks(theme='JohnSmith9982/small_and_pretty') as demo:
     gr.HTML("""<h1 style="text-align:center">Reversed Zotero</h1>""")
     text_input = gr.Textbox(label="Your text", type="text", lines=10)
     text_button = gr.Button("Process Text")
-    bibtex_output = gr.Textbox(label="BibTeX Entries", lines=15)
+    bibtex_output = gr.HTML(label="BibTeX Entries", lines=15)
     
     export_button = gr.Button("Export BibTeX")
     export_output = gr.File(label="Exported BibTeX File")