Create app.py

app.py

@@ -0,0 +1,90 @@
+import gradio as gr
+import threading
+import time
+from datetime import datetime
+from report_generator import generate_csv_report, generate_xlsx_report, generate_pdf_report
+
+def background_heartbeat():
+    """A continuous background task that logs a heartbeat every 60 seconds."""
+    while True:
+        print(f"[{datetime.now()}] Heartbeat: background task active.")
+        time.sleep(60)
+
+def process_command(user_input, chat_history):
+    """
+    Process the user command and perform tasks accordingly.
+    
+    Commands:
+      - If the text includes 'csv', generate a CSV report.
+      - If the text includes 'xlsx', generate an XLSX report.
+      - If the text includes 'pdf', generate a PDF report.
+      - Otherwise, echo the user's input with a friendly response.
+    
+    Returns:
+      - Updated chat history (list of tuples).
+      - A list of generated file paths (if any).
+    """
+    response = ""
+    file_links = []
+    lower_input = user_input.lower()
+    if "csv" in lower_input:
+        filename = generate_csv_report()
+        response = "CSV report generated."
+        file_links.append(filename)
+    elif "xlsx" in lower_input:
+        filename = generate_xlsx_report()
+        response = "XLSX report generated."
+        file_links.append(filename)
+    elif "pdf" in lower_input:
+        filename = generate_pdf_report()
+        response = "PDF report generated."
+        file_links.append(filename)
+    else:
+        response = f"You said: {user_input}. How can I help further?"
+
+    chat_history.append((user_input, response))
+    return chat_history, file_links
+
+def chat_interface(user_input, history):
+    """
+    The main function that gets called when the user submits a command.
+    It updates the conversation history and returns any generated files.
+    """
+    chat_history, file_links = process_command(user_input, history)
+    # Return updated conversation, the generated file(s), and update the state.
+    return chat_history, file_links, chat_history
+
+def main():
+    # Start the background heartbeat thread (daemonized so it exits when the main app stops)
+    heartbeat_thread = threading.Thread(target=background_heartbeat, daemon=True)
+    heartbeat_thread.start()
+
+    with gr.Blocks() as demo:
+        gr.Markdown("# Interactive Chatbot and Report Generator")
+        gr.Markdown(
+            "Type commands into the textbox below. For example, try: "
+            "`generate csv report`, `generate xlsx report`, or `generate pdf report`."
+        )
+        
+        with gr.Row():
+            chatbot = gr.Chatbot(label="Chatbot Conversation")
+        with gr.Row():
+            txt = gr.Textbox(label="Enter Command", placeholder="Type your command here...")
+        with gr.Row():
+            # This component will display generated files as downloadable links.
+            file_output = gr.File(label="Generated Files", file_count="multiple")
+        
+        # A hidden state to store conversation history.
+        state = gr.State([])
+
+        # When the user submits a command, update the chatbot, file output, and state.
+        txt.submit(
+            chat_interface,
+            inputs=[txt, state],
+            outputs=[chatbot, file_output, state]
+        ).then(lambda _: "", None, txt)
+
+    demo.launch(server_name="0.0.0.0", server_port=7860)
+
+if __name__ == "__main__":
+    main()