Update app.py

Update app to include graphic response

app.py

@@ -3,13 +3,17 @@ from huggingface_hub import CommitScheduler
 from datasets import Dataset
 import gradio as gr
 import pandas as pd
+import plotly.graph_objects as go
 import os
 
-from parameter_extractor import ParameterExtractor
 from DNAseq import DNAseq
+from grapher import DNAgrapher
+from parameter_extractor import ParameterExtractor
+
 from helper import list_at_index_0, list_at_index_1
 from logger import cts_log_file_create, logger, cts_logger
 
+
 HF_TOKEN = os.environ.get("HF_TOKEN", None)
 
 # Create csv file for data logging
@@ -29,7 +33,6 @@ scheduler = CommitScheduler(
 
 def chat_to_sequence(sequence, user_query):
     
-
     # Sequence to be analysed/queried
     input_sequence = sequence
 
@@ -48,8 +51,11 @@ def chat_to_sequence(sequence, user_query):
     # Set ParameterExtractor class expected variable
     query = user_query
 
-    # Bot Response
-    response = ""
+    # Initialise Graphic Response
+    fig = None
+
+    # Initialise Text Response
+    response = None
 
     # Query Code Description Message
     code_descript_message = ''
@@ -151,7 +157,13 @@ def chat_to_sequence(sequence, user_query):
         # Check if there is a match
         if not matching_row.empty:
             function = matching_row.iloc[0]["function"]
-            response = str(eval(function))
+            f_response = eval(function)
+            if code[0] == 'c':
+                response = None
+                fig = go.Figure(f_response)
+            else:
+                response = str(f_response)
+                fig = None
             code_descript_message = query_code_description.title()
             cts_logger(scheduler, log_file_path, log_data, response)
         else:
@@ -160,8 +172,8 @@ def chat_to_sequence(sequence, user_query):
             cts_logger(scheduler, log_file_path, log_data, response)
             print("No matching code found for the function:", code)
 
-        return response, code_descript_message
-    return response, code_descript_message
+        return response, fig, code_descript_message
+    return response, fig, code_descript_message
 
 
 ChatToSequence = gr.Interface(
@@ -184,6 +196,8 @@ ChatToSequence = gr.Interface(
          "What are the bases from position 2 to 10"],
         ["ggcattgaggagaccattgacaccgtcattagcaatgcactacaactgtcacaacctaaa",
          "How many bases are there from position 2 to 10"],
+        ["ggcattgaggagaccattgacaccgtcattagcaatgcactacaactgtcacaacctaaa",
+         "Show pie chart of total bases"],
     ],
 ).queue()