fix: add file content to chat state

app.py

@@ -93,16 +93,11 @@ def update_state(history: HistoryType, message: Optional[Dict[str, str]]):
     if isinstance(message, dict) and "files" in message:
         for file_path in message["files"]:
             try:
-                state.context = load_paper_as_context(file_path=file_path)
-                doc_context = [x.get_content() for x in state.context]
-                history.append(
-                    {
-                        "role": "assistant",
-                        "content": " ".join(doc_context)[
-                            : pipe.model.config.max_position_embeddings
-                        ],
-                    }
-                )
+                text = load_paper_as_context(file_path=file_path)
+                doc_context = [x.get_content() for x in text]
+                state.context = " ".join(doc_context)[
+                    : pipe.model.config.max_position_embeddings
+                ]
             except Exception as e:
                 history.append(
                     {"role": "assistant", "content": f"Error loading file: {str(e)}"}
@@ -149,47 +144,46 @@ def bot(history: HistoryType):
 
 def create_interface():
     with gr.Blocks() as demo:
-        with weave.attributes({"session": "hf-app"}):
-            global state
-            state = ChatState()
-            gr.Markdown(
-                """
-                <a href="https://github.com/SauravMaheshkar/papersai">
-                    <div align="center"><h1>papers.ai</h1></div>
-                </a>
-                """,
-            )
-            chatbot = gr.Chatbot(
-                show_label=False,
-                height=600,
-                type="messages",
-                show_copy_all_button=True,
-                placeholder="Upload a research paper and ask questions!!",
-            )
-
-            chat_input = gr.MultimodalTextbox(
-                interactive=True,
-                file_count="single",
-                placeholder="Upload a document or type your message...",
-                show_label=False,
-            )
-
-            chat_msg = chat_input.submit(
-                fn=update_state,
-                inputs=[chatbot, chat_input],
-                outputs=[chatbot, chat_input],
-            )
-
-            bot_msg = chat_msg.then(  # noqa: F841
-                fn=bot, inputs=[chatbot], outputs=chatbot, api_name="bot_response"
-            )
-
-            chatbot.like(
-                fn=record_feedback,
-                inputs=None,
-                outputs=None,
-                like_user_message=True,
-            )
+        global state
+        state = ChatState()
+        gr.Markdown(
+            """
+            <a href="https://github.com/SauravMaheshkar/papersai">
+                <div align="center"><h1>papers.ai</h1></div>
+            </a>
+            """,
+        )
+        chatbot = gr.Chatbot(
+            show_label=False,
+            height=600,
+            type="messages",
+            show_copy_all_button=True,
+            placeholder="Upload a research paper and ask questions!!",
+        )
+
+        chat_input = gr.MultimodalTextbox(
+            interactive=True,
+            file_count="single",
+            placeholder="Upload a document or type your message...",
+            show_label=False,
+        )
+
+        chat_msg = chat_input.submit(
+            fn=update_state,
+            inputs=[chatbot, chat_input],
+            outputs=[chatbot, chat_input],
+        )
+
+        bot_msg = chat_msg.then(  # noqa: F841
+            fn=bot, inputs=[chatbot], outputs=chatbot, api_name="bot_response"
+        )
+
+        chatbot.like(
+            fn=record_feedback,
+            inputs=None,
+            outputs=None,
+            like_user_message=True,
+        )
 
     return demo