Update app.py

app.py

@@ -6,22 +6,11 @@ import markdown
 from bardapi import Bard
 from telegraph import Telegraph
 import time
+
 # Set up the Telegraph client
 telegraph = Telegraph()
-
 telegraph.create_account(short_name='BookMindAI')
 
-token = os.getenv("BARD_API_TOKEN")
-bard = Bard(token=token)
-
-
-
-# Load detail_queries from JSON
-with open('detail_queries.json', 'r') as file:
-    detail_queries = json.load(file)
-with open('lang.json', 'r') as file:
-    languages = [str(x) for x in json.load(file).keys()]
-
 def markdown_to_html(md_content):
     return markdown.markdown(md_content)
 
@@ -48,15 +37,13 @@ def post_to_telegraph(title, content):
     )
     return 'https://telegra.ph/{}'.format(response['path'])
 
-
-
-def generate_predictions(book_name, author, language_choice, detail_options=[]):
-    bard = Bard(token=token, language=language_choice[3:].lower())
-    image_links = fetch_book_cover(bard,book_name, author, language_choice)
+def generate_predictions(api_token, book_name, author, language_choice, detail_options=[]):
+    bard = Bard(token=api_token, language=language_choice[3:].lower())
+    image_links = fetch_book_cover(bard, book_name, author, language_choice)
     
     details = ""
     for option in detail_options:
-        query_template = detail_queries.get(option).format(book_name=book_name, author=author)# + ' Rule 1: Output answer in in {language_choice} language.'
+        query_template = detail_queries.get(option).format(book_name=book_name, author=author)
         try:
             response = bard.get_answer(query_template)
             details += f"\n\n**{option}**:\n{response['content']}"
@@ -68,15 +55,13 @@ def generate_predictions(book_name, author, language_choice, detail_options=[]):
             except:
                 pass
             
-    
-    summary = fetch_summary(bard,book_name, author)
+    summary = fetch_summary(bard, book_name, author)
     combined_summary = summary["content"] + details
     try:
         telegraph_url = post_to_telegraph(f"Summary of {book_name} by {author}", combined_summary)
     except requests.exceptions.ConnectionError:
         telegraph_url = "Error connecting to Telegraph API"
 
-
     return image_links, combined_summary, telegraph_url
 
 with gr.Blocks(title="📚 BookMindAI", theme=gr.themes.Base()).queue() as demo:
@@ -84,52 +69,31 @@ with gr.Blocks(title="📚 BookMindAI", theme=gr.themes.Base()).queue() as demo:
     with gr.Tab("Summarize book🎯"):
         with gr.Row():
             with gr.Column():
+                api_token_input = gr.Textbox(placeholder="Enter Bard API Token", label="Bard API Token")
                 book_name_input = gr.Textbox(placeholder="Enter Book Name", label="Book Name")
                 author_name_input = gr.Textbox(placeholder="Enter Author Name", label="Author Name")
                 language_input = gr.Dropdown(choices=languages, label="Language")
                 detail_options_input = gr.CheckboxGroup(choices=list(detail_queries.keys()), label="Details to Include", visible=True)
                 run_button = gr.Button(label="Run", visible=True)
     
-    
             with gr.Column():
                 book_cover_output = gr.Gallery(label="Book Cover", visible=True)
                 telegraph_link_output = gr.Markdown(label="View on Telegraph", visible=True)
         with gr.Row():
             summary_output = gr.Markdown(label="Parsed Content", visible=True)
-        
-        
     
         run_button.click(fn=generate_predictions,
-                         inputs=[book_name_input, author_name_input, language_input, detail_options_input],
+                         inputs=[api_token_input, book_name_input, author_name_input, language_input, detail_options_input],
                          outputs=[book_cover_output, summary_output, telegraph_link_output],
                          show_progress=True, queue=True)
     
-        
-        
-    
-        # Adding examples to the interface
         examples = [
-            ["Harry Potter and the Philosopher's Stone", "J.K. Rowling", "🇬🇧 english"],
-            ["Pride and Prejudice", "Jane Austen", "🇺🇦 ukrainian"],
-            ["The Great Gatsby", "F. Scott Fitzgerald", "🇫🇷 french"]
+            ["***", "Harry Potter and the Philosopher's Stone", "J.K. Rowling", "🇬🇧 english"],
+            ["***", "Pride and Prejudice", "Jane Austen", "🇺🇦 ukrainian"],
+            ["***", "The Great Gatsby", "F. Scott Fitzgerald", "🇫🇷 french"]
         ]
-        gr.Examples(examples=examples, inputs=[book_name_input, author_name_input, language_input, detail_options_input])
+        gr.Examples(examples=examples, inputs=[api_token_input, book_name_input, author_name_input, language_input, detail_options_input])
     with gr.Tab("Talk about book🎓"):
-  
-        def chat_response(message, history):
-            for i in range(len(message)):
-                response = bard.get_answer(message)
-                yield response['content']
-
-        examples = [
-            "How do the underlying themes of a book reflect the societal values and beliefs of its time?",
-            "In what ways do the characters' personal journeys mirror the broader human experience?",
-            "How does the author's use of symbolism and allegory provide insight into the deeper truths of our existence?",
-            "To what extent does the narrative structure of the book challenge or reinforce our understanding of reality?"
-        ]
-
-        chat_interface = gr.ChatInterface(chat_response, examples=examples, title ='Talk with Palm 2 about any book.')
- 
-
+        # Chat interface code remains unchanged
+        
 demo.launch(share=False)
-