Update app.py

app.py

@@ -110,6 +110,9 @@ def split_text(text, max_tokens=500):
 
     return chunks
 
+def remove_punctuation(text):
+    return re.sub(r'[^\w\s]', '', text)
+
 def extract_year(text):
     year_match = re.search(r'\b(\d{4})\b', text)
     return year_match.group(1) if year_match else None
@@ -133,10 +136,13 @@ def create_bibtex_entry(data):
     author_words = data.get('author', '').split()
     first_author = author_words[0] if author_words else 'Unknown'
     bibtex_id = f"{first_author}{year}" if year else first_author
+    bibtex_id = remove_punctuation(bibtex_id.lower())
 
     bibtex = f"@{entry_type}{{{bibtex_id},\n"
     for key, value in data.items():
         if value.strip():
+            if key in ['volume', 'pages', 'year']:
+                value = remove_punctuation(value)
             bibtex += f"  {key.lower()} = {{{value.strip()}}},\n"
     bibtex = bibtex.rstrip(',\n') + "\n}"
     return bibtex
@@ -210,8 +216,7 @@ class MistralChatBot:
         print("BibTeX Entry:")
         print(bibtex_entry)
         
-        generated_text = f'{css}<h2 style="text-align:center">Edited text</h2>\n<div class="generation">{html_output}</div>'
-        return generated_text, bibtex_entry
+        return bibtex_entry
 
 # Create the Falcon chatbot instance
 mistral_bot = MistralChatBot()
@@ -232,9 +237,8 @@ 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=5)
     text_button = gr.Button("Extract a structured bibtex")
-    text_output = gr.HTML(label="Metadata")
     bibtex_output = gr.Textbox(label="BibTeX Entry", lines=10)
-    text_button.click(mistral_bot.predict, inputs=text_input, outputs=[text_output, bibtex_output])
+    text_button.click(mistral_bot.predict, inputs=text_input, outputs=[bibtex_output])
 
 if __name__ == "__main__":
     demo.queue().launch()