Deadmon commited on
Commit
f0a3c4e
·
verified ·
1 Parent(s): 38f85af

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -4
app.py CHANGED
@@ -8,6 +8,7 @@ import sqlite3
8
  import datetime
9
  import difflib
10
  import logging
 
11
  from tiktoken import get_encoding
12
  from openai import AzureOpenAI
13
  import httpx
@@ -362,7 +363,11 @@ async def chat_submit(user_input, chat_history, preprompt):
362
 
363
  def get_history():
364
  memory = ConversationMemory()
365
- return memory.get_recent_chunks(limit=10)
 
 
 
 
366
 
367
  def get_logs():
368
  try:
@@ -375,8 +380,8 @@ def get_logs():
375
  return f"Error reading logs: {str(e)}"
376
 
377
  def select_chunk(evt: gr.SelectData):
378
- logger.info(f"Selected chunk: {evt.value['chunk_id']}")
379
- return evt.value["chunk_id"], evt.value["text"]
380
 
381
  async def edit_cut(chunk_id, start, end):
382
  api = OpenAIApi(api_key=os.getenv("AZURE_OPENAI_API_KEY"))
@@ -423,7 +428,8 @@ def create_ui():
423
  history = gr.Dataframe(
424
  label="Recent Chunks",
425
  headers=["chunk_id", "text", "role", "timestamp", "intent", "token_count"],
426
- datatype=["str", "str", "str", "str", "str", "number"]
 
427
  )
428
  history_btn = gr.Button("Refresh History")
429
  history_btn.click(fn=get_history, outputs=history)
 
8
  import datetime
9
  import difflib
10
  import logging
11
+ import pandas as pd
12
  from tiktoken import get_encoding
13
  from openai import AzureOpenAI
14
  import httpx
 
363
 
364
  def get_history():
365
  memory = ConversationMemory()
366
+ chunks = memory.get_recent_chunks(limit=10)
367
+ # Convert to list of lists for Gradio Dataframe
368
+ data = [[chunk["chunk_id"], chunk["text"], chunk["role"], chunk["timestamp"], chunk["intent"], chunk["token_count"]] for chunk in chunks]
369
+ logger.info(f"Returning {len(data)} chunks for history: {json.dumps(data, ensure_ascii=False)}")
370
+ return data
371
 
372
  def get_logs():
373
  try:
 
380
  return f"Error reading logs: {str(e)}"
381
 
382
  def select_chunk(evt: gr.SelectData):
383
+ logger.info(f"Selected chunk: {evt.value[0]}")
384
+ return evt.value[0], evt.value[1]
385
 
386
  async def edit_cut(chunk_id, start, end):
387
  api = OpenAIApi(api_key=os.getenv("AZURE_OPENAI_API_KEY"))
 
428
  history = gr.Dataframe(
429
  label="Recent Chunks",
430
  headers=["chunk_id", "text", "role", "timestamp", "intent", "token_count"],
431
+ datatype=["str", "str", "str", "str", "str", "number"],
432
+ interactive=False
433
  )
434
  history_btn = gr.Button("Refresh History")
435
  history_btn.click(fn=get_history, outputs=history)