Update app.py

app.py

@@ -25,8 +25,6 @@ bibliography_classifier = pipeline(
     "token-classification", model=bibliography_model, aggregation_strategy="simple", device=device
 )
 
-style_classifier = pipeline("text-classification", model=bibliography_style, tokenizer=tokenizer, device=device)
-
 # Helper functions
 def preprocess_text(text):
     text = re.sub(r'<[^>]+>', '', text)
@@ -179,8 +177,8 @@ class CombinedProcessor:
             print(entry)
             entry = re.sub(r'- ?[\n¶] ?', r'', entry)
             entry = re.sub(r' ?[\n¶] ?', r' ', entry)
-            style = pd.DataFrame(style_classifier(entry, truncation=True, padding=True, top_k=1))
-            list_style.append(style)
+            #style = pd.DataFrame(style_classifier(entry, truncation=True, padding=True, top_k=1))
+            #list_style.append(style)
             entry = re.sub(r'\s*([;:,\.])\s*', r' \1 ', entry)
             #print(entry)
             bib_out = bibliography_classifier(entry)
@@ -207,18 +205,19 @@ class CombinedProcessor:
             bibtex_entry = create_bibtex_entry(bibtex_data)
             bibtex_entries.append(bibtex_entry)
 
-        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']
-        top_style_score = list_style.iloc[0]['score']
+        #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']
+        #top_style_score = list_style.iloc[0]['score']
 
         # Create the style information string
-        style_info = f"Top bibliography style: {top_style} (Mean score: {top_style_score:.6f})"
+        #style_info = f"Top bibliography style: {top_style} (Mean score: {top_style_score:.6f})"
 
         # Join BibTeX entries
         bibtex_content = "\n\n".join(bibtex_entries)
 
-        return style_info, bibtex_content
+        #return style_info, bibtex_content
+        return bibtex_content
 
 # Create the processor instance
 processor = CombinedProcessor()
@@ -228,13 +227,12 @@ 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")
-    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=[style_output, bibtex_output])
+    text_button.click(processor.process, inputs=text_input, outputs=[bibtex_output])
     export_button.click(save_bibtex, inputs=[bibtex_output], outputs=[export_output])
 
 if __name__ == "__main__":