Update app.py

app.py

@@ -148,9 +148,7 @@ def create_bibtex_entry(data):
     bibtex = bibtex.rstrip(',\n') + "\n}"
     return bibtex
 
-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)
+def save_bibtex(bibtex_content):
     with tempfile.NamedTemporaryFile(mode='w', delete=False, suffix='.bib') as temp_file:
         temp_file.write(bibtex_content)
     return temp_file.name
@@ -212,22 +210,15 @@ class CombinedProcessor:
         list_style = pd.concat(list_style)
         list_style = list_style.groupby('label')['score'].mean().sort_values(ascending=False).reset_index()
         top_style = list_style.iloc[0]['label']
-        print(top_style)
-        
-        #Disambiguation to avoid duplicate ids.
-        bibtex_entries = disambiguate_bibtex_ids(bibtex_entries)
-        
-        # Join BibTeX entries with HTML formatting
-        formatted_entries = [html.escape(entry) for entry in bibtex_entries]
+        top_style_score = list_style.iloc[0]['score']
 
-        bibtex_content = "\n\n".join(formatted_entries)
+        # Create the style information string
+        style_info = f"Top bibliography style: {top_style} (Mean score: {top_style_score:.6f})"
 
-        # Create a paragraph with the top style information
-        style_info = f"<p>Estimated bibliographic style: {top_style}</p>"
+        # Join BibTeX entries
+        bibtex_content = "\n\n".join(bibtex_entries)
 
-        # Combine the style information and BibTeX content
-        result = f"{style_info}\n\n{bibtex_content}"
-        return result
+        return style_info, bibtex_content
 
 # Create the processor instance
 processor = CombinedProcessor()
@@ -237,12 +228,13 @@ 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.HTML(label="BibTeX Entries")
+    style_output = gr.Textbox(label="Top Style", lines=2)
+    bibtex_output = gr.Textbox(label="BibTeX Entries", lines=15)
     
     export_button = gr.Button("Export BibTeX")
     export_output = gr.File(label="Exported BibTeX File")
 
-    text_button.click(processor.process, inputs=text_input, outputs=[bibtex_output])
+    text_button.click(processor.process, inputs=text_input, outputs=[style_output, bibtex_output])
     export_button.click(save_bibtex, inputs=[bibtex_output], outputs=[export_output])
 
 if __name__ == "__main__":